from pydantic import BaseModel
from typing import List, Optional
from fastapi import UploadFile

class OCRRequest(BaseModel):
    query_keys: List[str]   # 需要提取的关键字列表（必填）
    use_orientation: bool = False  # 是否启用方向分类（可选）
    use_unwarping: bool = False     # 是否启用文档矫正（可选）

class OCRResponse(BaseModel):
    result: dict            # 问答结果
    message: str = "success"
    code: int = 200